/*
 * Copyright (c) 2025 DuckDuckGo
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.duckduckgo.pir.impl.store.db

import androidx.room.Dao
import androidx.room.Insert
import androidx.room.OnConflictStrategy
import androidx.room.Query
import kotlinx.coroutines.flow.Flow

@Dao
interface ExtractedProfileDao {
    @Query("SELECT * FROM pir_extracted_profiles ORDER BY dateAddedInMillis")
    fun getAllExtractedProfileFlow(): Flow<List<StoredExtractedProfile>>

    @Query("SELECT * FROM pir_extracted_profiles ORDER BY dateAddedInMillis")
    fun getAllExtractedProfiles(): List<StoredExtractedProfile>

    @Query("SELECT * FROM pir_extracted_profiles WHERE id = :id ORDER BY dateAddedInMillis")
    fun getExtractedProfile(id: Long): StoredExtractedProfile?

    @Query("SELECT * FROM pir_extracted_profiles WHERE profileQueryId = :profileQueryId ORDER BY dateAddedInMillis")
    fun getExtractedProfilesForProfile(profileQueryId: Long): List<StoredExtractedProfile>

    @Query("SELECT * FROM pir_extracted_profiles WHERE brokerName = :brokerName ORDER BY dateAddedInMillis")
    fun getExtractedProfilesForBroker(brokerName: String): List<StoredExtractedProfile>

    @Query("SELECT * FROM pir_extracted_profiles WHERE brokerName = :brokerName AND profileQueryId = :profileQueryId ORDER BY dateAddedInMillis")
    fun getExtractedProfilesForBrokerAndProfile(
        brokerName: String,
        profileQueryId: Long,
    ): List<StoredExtractedProfile>

    @Insert(onConflict = OnConflictStrategy.IGNORE)
    fun insertNewExtractedProfile(extractedProfile: StoredExtractedProfile)

    @Insert(onConflict = OnConflictStrategy.IGNORE)
    fun insertNewExtractedProfiles(extractedProfiles: List<StoredExtractedProfile>)

    @Query("DELETE from pir_extracted_profiles")
    fun deleteAllExtractedProfiles()

    @Query("UPDATE pir_extracted_profiles SET deprecated = :deprecated WHERE id = :extractedProfileId")
    fun updateExtractedProfileDeprecated(
        extractedProfileId: Long,
        deprecated: Boolean,
    )
}
